import mysql.connector
from mysql.connector import Error

def check_mysql_connection():
    """
    检查MySQL连接并创建数据库（如果不存在）
    """
    try:
        # 连接到MySQL服务器（不指定数据库）
        connection = mysql.connector.connect(
            host="localhost",
            user="root",
            password="Chyxwt123@"
        )

        if connection.is_connected():
            db_info = connection.get_server_info()
            print(f"已连接到MySQL服务器，版本: {db_info}")

            # 创建游标对象
            cursor = connection.cursor()

            # 检查数据库是否存在
            cursor.execute("SHOW DATABASES LIKE 'django_exam_db'")
            result = cursor.fetchone()

            if result:
                print("数据库 'django_exam_db' 已存在")
            else:
                # 创建数据库
                cursor.execute("CREATE DATABASE django_exam_db CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci")
                print("数据库 'django_exam_db' 已创建")

            cursor.close()
            connection.close()
            print("MySQL连接已关闭")
            return True

    except Error as e:
        print(f"连接MySQL时出错: {e}")
        return False

if __name__ == "__main__":
    check_mysql_connection()